package com.its.util;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTCreator;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.DecodedJWT;

import java.util.Calendar;
import java.util.Map;

/**
 * JWT工具类用于生成和验证Token
 * @author 28691
 */
public class JWTUtils {
    private static final String SECRET = "token!Q@W#E$R"; // 密钥，用于签名和验证Token

    /**
     * 生成Token
     * @param map 存储在Token中的payload信息
     * @return 生成的Token字符串
     */
    public static String getToken(Map<String, String> map) {
        JWTCreator.Builder builder = JWT.create();

        // 设置payload信息
        map.forEach(builder::withClaim);

        // 默认设置Token过期时间为7天
        Calendar instance = Calendar.getInstance();
        instance.add(Calendar.DATE, 7);
        builder.withExpiresAt(instance.getTime()); // 指定令牌的过期时间

        // 生成Token并进行签名
        return builder.sign(Algorithm.HMAC256(SECRET));
    }

    /**
     * 验证Token的有效性
     * @param token 待验证的Token
     * @return 解析后的Token信息
     */
    public static DecodedJWT verify(String token) {
        // 验证Token的签名和有效性，如果验证失败会抛出异常
        return JWT.require(Algorithm.HMAC256(SECRET)).build().verify(token);
    }
}
